from django.test import TestCase

# Create your tests here.
from django.test import TestCase
from llm.models import VectorStores


class YourModelTest(TestCase):

    def setUp(self):
        # 创建一个测试对象
        VectorStores.objects.get(user_id_id="131dbd028a5a4aefa6aee02db649ab24")
        pass

    def test_model_str(self):
        # 测试模型的__str__方法
        obj = VectorStores.objects.get(user_id_id="131dbd028a5a4aefa6aee02db649ab24")
        print(obj)

    def test_os(self):
        import os
        print(os.getenv("OPENAI_API_KEY"))
